#ifndef __UTIL__HPP__
#define __UTIL__HPP__

#include "Log.hpp"
#include <fstream>
#include <memory>
#include <vector>
#include <unordered_map>
#include <mutex>
#include <mysql/mysql.h>
#include <jsoncpp/json/json.h>

#include <websocketpp/server.hpp>
#include <websocketpp/config/asio_no_tls.hpp>

typedef websocketpp::server<websocketpp::config::asio> wsserver_t;

/*工具类方法对外开放 , 供外部使用*/

class mysql_util
{
public:
    /*static 成员函数外部不用实例化*/
    static MYSQL *mysql_create(const std::string &host, const std::string &user,
                               const std::string &passwd, const std::string &dbname,
                               unsigned int port = 3306)
    {
        /*1.初始化mysql*/
        MYSQL *sql = mysql_init(nullptr);
        /*2.连接数据库*/
        MYSQL *mysql = mysql_real_connect(sql, host.c_str(),
                                          user.c_str(),
                                          passwd.c_str(),
                                          dbname.c_str(),
                                          port, nullptr, 0);
        if (mysql == nullptr)
        {
            FALOG("mysql connect fail %s", mysql_error(mysql));
            mysql_close(mysql);
            return nullptr;
        }
        /*3.设置字符*/
        if (mysql_set_character_set(mysql, "utf8") != 0)
        {
            ERRLOG("for mysql set character_set failed !");
            mysql_close(mysql);
            return nullptr;
        }

        return mysql;
    }

    static bool SqlExecute(MYSQL *mysql, const std::string &sql)
    {
        if (mysql_query(mysql, sql.c_str()) != 0)
        {
            ERRLOG("SQL [%s] execute failed !", sql.c_str());
            ERRLOG("ERR : %s", mysql_error(mysql));
            return false;
        }
        return true;
    }

    static void mysql_destory(MYSQL *msq)
    {
        if (msq != nullptr)
        {
            mysql_close(msq);
        }
    }
};

/*提供序列化和反序列化接口*/
class json_util
{
public:
    /*序列化*/
    static bool Serialize(Json::Value &root, std::string &str)
    {
        /*1.构建一个 StreamWriteBuilde 工厂类对象*/
        Json::StreamWriterBuilder swb;
        // 全局默认启用UTF-8编码，中文将直接显示
        swb["emitUTF8"] = true;
        /*2.用 StreamWriteBuilde 工厂类生产一个 StreamWrite对象*/
        std::unique_ptr<Json::StreamWriter> sw(swb.newStreamWriter());
        /*3. 将 root 内容序列化成字符串*/
        std::stringstream ss;
        if (sw->write(root, &ss) != 0)
        {
            ERRLOG("Serialize falied !");
            return false;
        }
        str = ss.str();
        return true;
    }

    /*反序列化*/
    static bool Unserialize(std::string &str, Json::Value &root)
    {
        /*1.构建一个 CharReaderBuild 工厂类对象*/
        Json::CharReaderBuilder crb;
        /*2.用 CharReaderBuild 工厂类生产一个 CharReader 对象*/
        std::unique_ptr<Json::CharReader> cr(crb.newCharReader());
        /*3.将 str 反序列化 存储在 root */
        bool ret = cr->parse(str.c_str(), str.c_str() + str.size(), &root, nullptr);
        if (!ret)
        {
            ERRLOG("%s Unserialize failed !", str.c_str());
            return false;
        }
        return true;
    }
};

/*提供字符串分割功能*/
class string_util
{
public:
    static int split(const std::string &src, const std::string &sep, std::vector<std::string> &res)
    {
        /*11111,2424,1,4,,,,,51,,*/
        int index = 0;
        while (index < src.size())
        {
            auto pos = src.find(sep, index);
            if (pos == std::string::npos) /*1111*/
            {
                std::string str = src.substr(index);
                res.emplace_back(str);
                break;
            }
            /*,,,,*/
            if (index == pos)
            {
                index += sep.size();
                continue;
            }
            /*11111,2424,1,4,,,,,51,,*/
            std::string sepstring = src.substr(index, pos - index); /*11111*/
            res.emplace_back(sepstring);
            index = pos + sep.size();
        }

        return res.size();
    }
};

/*提供文件工具 , 读取服务器上的文件*/
class file_util
{
public:
    static bool read(const std::string& filename , std::string& body)
    {
        std::ifstream ifs(filename , std::ios::binary); /*默认就是读打开 , 这里 + 二进制*/
        if(!ifs.is_open())
        {
            FALOG("open %s failed !", filename.c_str());
            return false;
        }

        //获取文件总大小
        /*1. 将文件指针移动到末尾 , 就是文件大小*/
        ifs.seekg(0,std::ios::end);
        /*2. 获取文件大小*/
        size_t fsize = ifs.tellg();
        /*3. 恢复到文件开头,准备读取*/
        ifs.seekg(0,std::ios::beg);
        body.resize(fsize);

        //读取 ,将文件数据读取到 body 中
        ifs.read(&body[0] , fsize); /* 注意 :  read 第一个参数为 char*(指向文件首地址) 只能用 &body[0]*/ /*不能 filename.c_str() 因为其是 const char* */
        if(!ifs.good())
        {
            ERRLOG("read %s failed !" , filename.c_str());
            ifs.close();
            return false;
        }

        ifs.close();
        return true;
    }
};
#endif

/**
 * 本 hpp 实现了项目中用到的相关实用工具 , 对工具基本功能封装
 * MySQL 数据库 : 初始化 , 执行句柄 , 销毁句柄
 * Json : 序列化和反序列化 , 为通信做准备
 * StringSep : 为后续分割字符串做准备 , 关注 : 怎么分割的 , 出现重复字符怎么办 ?
 * File : 读取服务器文件时使用 , 因为网页页面本质就是服务器上的一个文件 , 需要客户端读取获取 , 关注 : 获取文件大小用文件读写位置 ,先移动到末尾再恢复的思想
 * ! 以上功能均进行了 Debug 调试 , 功能均正常 !
 * ... 有的话在添加 ...
 */